package com.study.util;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

/**
 * @Author wfw
 * @Date 2020/07/11 18:00
 */
public class DownloadUtils {
    public void download(ByteArrayOutputStream byteArrayOutputStream, HttpServletResponse response, String returnName) throws IOException {
        response.setContentType("application/octet-stream");
        // 保存的文件名,必须和页面编码一致,否则乱码
        returnName = response.encodeURL(new String(returnName.getBytes(),"iso8859-1"));
        response.addHeader("content-disposition","attachment;filename=" + returnName);
        response.setContentLength(byteArrayOutputStream.size());
        // 取得输出流
        ServletOutputStream outputstream = response.getOutputStream();
        // 写到输出流
        byteArrayOutputStream.writeTo(outputstream);
        // 关闭
        byteArrayOutputStream.close();
        // 刷数据
        outputstream.flush();
    }
}
